package com.echo.im.common.util;

import org.springframework.util.StringUtils;

import java.util.HashSet;
import java.util.Set;

/**
 * @author echo
 */
public final class FileUtil {
    private static final Set<String> IMAGE_EXTENSIONS = new HashSet<>() {{
        add("jpeg");
        add("jpg");
        add("bmp");
        add("png");
        add("webp");
        add("gif");
    }};

    /**
     * 获取文件后缀
     *
     * @param fileName 文件名
     * @return boolean
     */
    public static String getFileExtension(String fileName) {
        return fileName.substring(fileName.lastIndexOf(".") + 1);
    }

    /**
     * 判断文件是否图片类型
     *
     * @param fileName 文件名
     * @return boolean
     */
    public static boolean isImage(String fileName) {
        String extension = getFileExtension(fileName);
        return StringUtils.hasText(extension) && IMAGE_EXTENSIONS.contains(extension.toLowerCase());
    }
}
